import knn_basic  
from numpy import *   
  
dataSet, labels = knn_basic.createDataSet()  
  
testX = array([1.2, 1.0])  
k = 3  
outputLabel = knn_basic.kNNClassify(testX, dataSet, labels, 3)  
print("Your input is:", testX, "and classified to class: ", outputLabel)
  
testX = array([0.1, 0.3])  
outputLabel = knn_basic.kNNClassify(testX, dataSet, labels, 3)
print("Your input is:", testX, "and classified to class: ", outputLabel)